package networks;

import java.io.IOException;
import java.net.*;
import java.util.Scanner;

public class UdpEchoClient {

    private DatagramSocket socket = null;

    // 由于UDP不存储对端的IP和端口，因此需要自行定义
    private String serverIP;
    private int serverPort;

    public UdpEchoClient(String serverIP, int serverPort) throws SocketException {
        this.serverIP = serverIP;
        this.serverPort = serverPort;

        // 这里的端口不能固定，因为客户端会运行其他的程序，如果是固定端口，当这个端口被占用时，就会导致这个程序运行失败
        socket = new DatagramSocket();
    }

    public void start() throws IOException {
        Scanner scanner = new Scanner(System.in);
        System.out.println("客户端程序启动...");

        while(true){
            // 1. 从控制台读取用户输入内容
            System.out.println("请输入要传输的内容：");
            String request = scanner.next();

            // 2. 把请求发给服务器
            //      构造过程中，不仅需要构造载荷，还需要服务器的ip和端口号
            //      由于这DatagramPacket没有适用String类型字符串的重载方法，而是通过InetAddress这个类来间接获取目的IP
            DatagramPacket requestPacket = new DatagramPacket(request.getBytes(), request.getBytes().length,
                    InetAddress.getByName(serverIP), serverPort);

            // 3. 发送数据报
            socket.send(requestPacket);

            // 4. 接收服务器的相应
            DatagramPacket responsePacket = new DatagramPacket(new byte[4096], 4096, InetAddress.getByName(serverIP), serverPort);
            socket.receive(responsePacket);

            // 5. 解析接收到的数据
            String response = new String(responsePacket.getData(), 0, responsePacket.getLength());
            System.out.println(response);
        }
    }

    public static void main(String[] args) throws IOException {
        UdpEchoClient client = new UdpEchoClient("127.0.0.1", 9090);
        client.start();
    }
}
